package _11_整理题目._3_字符串_递归_dfs_bfs_双指针._排列组合._基本_排列组合数;

import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

/**
 * Anm ：从 n 个不同元素中，取出 m 个元素 组成不同的排列 的个数
 *
 * 思路：dfs
 *      边界判断：当前 path.size() == m （只有边界条件变了）
 *          添加到结果集中，结束
 *      递归控制：for 遍历 0~n-1，使用 visited 控制不重复拿取
 *          如果 visited continue
 *          添加当前元素
 *          标记已访问当前元素
 *          dfs()
 *          撤销已访问当前元素
 *          删除当前元素
 */
public class _011_排列数_Anm {

    private List<List<Integer>> result = new ArrayList<>();
    private int k = 0;

    public List<List<Integer>> permute(int[] nums, int k) {
        this.k = k;
        List<Integer> path = new ArrayList<>();
        boolean[] visited = new boolean[nums.length];
        dfs(nums, path, visited);
        return result;
    }

    void dfs(int[] nums, List<Integer> path, boolean[] visited) {
        if (path.size() == k) {
            result.add(new ArrayList<>(path));
            return;
        }

        for (int i = 0; i < nums.length; i++) {
            if (visited[i]) continue;

            path.add(nums[i]);
            visited[i] = true;

            dfs(nums, path, visited);

            path.remove(path.size()-1);
            visited[i] = false;
        }
    }

    @Test
    public void main() {
        int[] nums = {1, 2, 3, 4};
        List<List<Integer>> res = permute(nums, 3);
        System.out.println("res.size() = " + res.size());
        // A43 = C43 * A33 = 4 * 3! = 24
        res.forEach(System.out::println);
    }
}
